from selenium import webdriver
import time

# 初始化Chrome浏览器
driver = webdriver.Chrome()

# 设置页面加载的超时时间（单位：秒）
driver.set_page_load_timeout(10)

try:
    # 访问百度首页
    driver.get("https://www.baidu.com")
    driver.maximize_window()

    print("页面加载成功")

    # 截图保存
    driver.save_screenshot("baidu_page_load.png")
    print("截图已保存")

except Exception as e:
    print(f"页面加载超时：{e}")

finally:
    # 关闭浏览器
    driver.quit()